# -*- coding:utf-8 -*-
from learning_agent import LearningAgent
from our_bag_of_words import OurBagOfWords
from nearest_neighbor import NearestNeighbor

class Gonzales(LearningAgent):
    
    def __init__(self, stemming = False,stopWords = False, nFeatures = 5):
        self.stemming = stemming
        self.stopWords = stopWords
        self.nFeatures = nFeatures
    
    def createFeatureExtractor(self):
        return OurBagOfWords(self.nFeatures, self.stemming, self.stopWords)
    
    def createClassifier(self):
        return NearestNeighbor()
    
    def __str__(self):
        return 'Gonzales'
    
